#ifndef __M_BUFFER_H__
#define __M_BUFFER_H__
/*实现异步日志缓冲区*/

#include "util.hpp"
#include <vector>
#include <cassert>
#include <algorithm>
namespace logslearn
{
// 定义宏，表示缓冲区的大小
#define DEFAULT_BUFFER_SIZE (1 * 1024 * 1024)
#define THRESHOLD_BUFFER_SIZE (8 * 1024 * 1024)
#define INCREMENT_BUFFER_SIZE (1 * 1024 * 1024)
    // 异步缓冲区
    class Buffer
    {
    public:
        // 构造函数
        Buffer() : _buffer(DEFAULT_BUFFER_SIZE), _reader_idx(0), _writer_idx(0) {}
        // 1.向缓冲区写入数据,容量不够就扩容（两种方式，极限测试的时候使用扩容，实际使用过程中固定空间大小，空间不够阻塞）
        void push(const char *data, size_t len)
        {
            // 缓冲区剩余空间不够的情况下：扩容。
            // // 1.固定大小,直接返回
            // if (len > writeAbleSize())
            //     return;
            // 2.动态空间，用于极限测试--扩容
            //空间不够则扩容
            ensureEnoughSize(len);

            // 2.将数据拷贝到缓冲区
            std::copy(data, data + len, &_buffer[_writer_idx]);
            // 3.将当前写入位置向后偏移
            moveWriter(len);
        }
        // 2.返回可读数据起始地址的接口
        const char *begin()
        {
            return &_buffer[_reader_idx];
        }
        // 3.返回可读数据的长度的接口；返回可写数据的长度的接口
        size_t readAbleSize()
        {
            // 返回可读数据的长度
            // 因为当前实现的缓冲区设计思想是双缓冲区，处理完就交换，所以不存在空间循环使用
            return (_writer_idx - _reader_idx);
        }
        size_t writeAbleSize()
        {
            // 对于扩容的思路来说，不存在可写空间大小，因为总是可写的。
            // 因此这个接口只提供给固定大小缓冲区。
            return (_buffer.size() - _writer_idx);
        }
        // 4.移动读写指针进行向后偏移的接口
        void moveWriter(size_t len)
        {
            // 移动可写指针
            assert((_writer_idx + len) <= _buffer.size());
            _writer_idx += len;
        }
        void moveReader(size_t len)
        {
            // 移动可读指针
            assert(len <= readAbleSize()); // len大就报错
            _reader_idx += len;
        }
        // 5.重置读写位置，初始化缓冲区的操作
        void reset()
        {
            // 读写位置都为零
            _writer_idx = 0; // 缓冲区所有空间都是空闲得
            _reader_idx = 0; //_reader_idx与_writer_idx相等就表示没有数据可以读
        }
        // 6.交换缓冲区的接口
        void swap(Buffer &buffer)
        {
            // 交换缓冲区地址
            _buffer.swap(buffer._buffer);
            // 读写位置的地址也要交换
            std::swap(_reader_idx, buffer._reader_idx);
            std::swap(_writer_idx, buffer._writer_idx);
        }
        // 判断缓冲区是否为空
        bool empty()
        {
            return (_reader_idx == _writer_idx); // 位置一样就是空
        }

    private:
        // 对空间进行扩容
        void ensureEnoughSize(size_t len)
        {
            // 不需要扩容
            if (len <= writeAbleSize())
                return;
            size_t new_size = 0;
            if (_buffer.size() < THRESHOLD_BUFFER_SIZE)
            {
                // 小于阈值则翻倍增长
                new_size = _buffer.size() * 2;
            }
            else
            {
                new_size = _buffer.size() + INCREMENT_BUFFER_SIZE+len; // 否则线性增长
            }
            // 重新调整空间大小
            _buffer.resize(new_size);
        }

    private:
        // 1.存放字符串数据的缓冲区
        std::vector<char> _buffer;
        // 2.当前可写数据的指针--本质是下标
        size_t _reader_idx;
        // 3.当前可读数据的指针
        size_t _writer_idx;
    };
}
#endif